from pydantic import BaseModel, EmailStr
from typing import Optional
from datetime import datetime

class ChatBase(BaseModel):
    chat_id: int


class SendMessageParam(ChatBase):
    problem_id: int #问题ID
    vueVirtualUUID: str #前端生成的消息uid
    content: str

class AddMessageParam(BaseModel):
    vueVirtualUUID: str #前端生成的消息uid
    ai_agent_id: int #智能体ID
    content: str
    chat_id: Optional[int] =None

class DeleteMessageParam(BaseModel):
    id: int

class ChangeChatAiAgent(BaseModel):
    chat_id: int
    ai_agent_id: int